/*
 *  Copyright 2013 National Institute of Advanced Industrial Science and Technology
 *  
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *  
 *      http://www.apache.org/licenses/LICENSE-2.0
 *  
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

package org.sss.mapreduce.example;

import org.sss.client.JobEngine;
import org.sss.client.SssClient;
import org.sss.mapreduce.GroupID;

public class WordCountEndless {
  /**
   * Usage:
   * java WordCount groupid nloop
   */
  public static void main(String[] args) throws Exception {
    SssClient client = new SssClient(args);

    String[] others = client.getArgs();
    if (others.length < 2) {
      System.err.println("input group id is not specified.");
      System.exit(1);
    }
    if (others.length > 3) {
      System.err.println("too many arguments.");
      System.exit(1);
    }
    String input = others[0];
    int nloop = Integer.parseInt(others[1]);

    JobEngine engine = new JobEngine(client);
    try {
      for (int i = 0;i < nloop;++i) {
        GroupID c0 = GroupID.createFromString(input);
        GroupID c1 = GroupID.createRandom(engine);
        GroupID c2 = GroupID.createRandom(engine);

        engine.getJobBuilder("jobA", TokenizerMapper.class)
          .addInput(c0) .addOutput(c1).build();
        engine.getJobBuilder("jobB", IntSumReducer.class)
          .addInput(c1).addOutput(c2).build();
        engine.exec();

        System.out.println("done");

        long n1 = client.removeTupleGroup(c1);
        System.out.format("Removed %d items(mapper output).\n", n1);
        long n2 = client.removeTupleGroup(c2);
        System.out.format("Removed %d items(reducer output).\n", n2);
      }
    } finally {
      engine.dispose();
    }
  }
}
